package cn.ynrcb.personnelfile.aspect;

import cn.ynrcb.personnelfile.common.annotation.SysLog;
import cn.ynrcb.personnelfile.common.utils.HttpContextUtils;
import cn.ynrcb.personnelfile.common.utils.IPUtils;
import cn.ynrcb.personnelfile.sysadmin.entity.SysLogEntity;
import cn.ynrcb.personnelfile.sysadmin.entity.SysUserEntity;
import cn.ynrcb.personnelfile.sysadmin.service.SysLogService;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.google.gson.Gson;
import org.apache.shiro.SecurityUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;
import java.util.Date;


/**
 * 系统日志，切面处理类 AOP
 * @author swb
 * @version created at 2020年7月31日 下午2:57:07
 */
@Aspect
@Component
public class SysLogAspect {
	@Autowired
	private SysLogService sysLogService;

	@Pointcut("@annotation(cn.ynrcb.personnelfile.common.annotation.SysLog)")
	public void logPointCut() {

	}

	@Around("logPointCut()")
	public Object around(ProceedingJoinPoint point) throws Throwable {
		long beginTime = System.currentTimeMillis();
		//执行方法
		Object result = point.proceed();
		//执行时长(毫秒)
		long time = System.currentTimeMillis() - beginTime;

		//保存日志
		saveSysLog(point, time);

		return result;
	}

	private void saveSysLog(ProceedingJoinPoint joinPoint, long time) {
		MethodSignature signature = (MethodSignature) joinPoint.getSignature();
		Method method = signature.getMethod();

		SysLogEntity sysLog = new SysLogEntity();
		SysLog syslog = method.getAnnotation(SysLog.class);
		if(syslog != null){
			//注解上的描述
			sysLog.setOperation(syslog.value());
		}

		//请求的方法名
		String className = joinPoint.getTarget().getClass().getName();
		String methodName = signature.getName();
		sysLog.setMethod(className + "." + methodName + "()");

		//请求的参数
		Object[] args = joinPoint.getArgs();
		try{
			String params = new Gson().toJson(args);
			sysLog.setParams(params);
		}catch (Exception e){

		}

		//获取request
		HttpServletRequest request = HttpContextUtils.getHttpServletRequest();
		//设置IP地址
		sysLog.setIp(IPUtils.getIpAddr(request));

		//用户名
		if(null != SecurityUtils.getSubject().getPrincipal()) {
			String username = ((SysUserEntity) SecurityUtils.getSubject().getPrincipal()).getUsername();
			sysLog.setUsername(username);
		}else {
			if(sysLog.getOperation().equals("用户登录")) {
				String username = JSONArray.parseArray(sysLog.getParams()).getJSONObject(0).getString("username");
				sysLog.setUsername(username);
			}
		}

		//删除密码记录
		if (sysLog.getOperation().equals("用户登录")
				|| sysLog.getOperation().equals("注册用户")) {
			if (sysLog.getParams().contains("password")) {
				JSONObject params = JSONArray.parseArray(sysLog.getParams()).getJSONObject(0);
				params.remove("password");
				sysLog.setParams(params.toJSONString());
			}
		} else if (sysLog.getOperation().equals("修改密码")) {
			sysLog.setParams(null);
		}

		sysLog.setTime(time);
		sysLog.setCreateDate(new Date());
		//保存系统日志
		sysLogService.save(sysLog);
	}
}
